星尘

349. 两个数组的交集

发表于 更新于 分类于

349. 两个数组的交集

题目

给定两个数组，编写一个函数来计算它们的交集。

示例 1:

1
2
输入：nums1 = [1,2,2,1], nums2 = [2,2]
输出：[2]

示例 2:

1
2
输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出：[9,4]

题解

map 具有去重的功能，每个元素记录一次，如果有交集则记录值 -1。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
class Solution(object):
    def intersection(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        # 方法1：直接用取交集的方法
        # 方法2：是使用map实现取交集
        result = []
        mapper = dict()
        for x in nums1:
            mapper[x] = 1
        for y in nums2:
            if mapper.get(y) and mapper.get(y) == 1:
                result.append(y)
                mapper[y] = 0
        return result
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
func intersection(nums1 []int, nums2 []int) []int {
	var mapper = make(map[int]int)
	var result []int

	for _, x:= range nums1{
		mapper[x] = 1
	}

	for _, y:= range nums2{
		i, ok := mapper[y]
		if ok && i == 1{
			result = append(result, y)
			mapper[y] = 0
		}
	}

	return result
}

变体

总结

参考